try:
    from .trainer import GANTrainer
except ImportError:
    from trainer import GANTrainer
